import random
from datetime import datetime, timedelta


# 更新今日巡检工单状态，处理时长随机为25-40分钟（直接改库）
# 数据库连接
from dao.mysqlhelper import MySQLHelper

iot=MySQLHelper()


# 获取当前时间
current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
current_day = datetime.now().strftime('%Y-%m-%d')

# 查询符合条件的行
select_query = f"SELECT wo_no, created_time FROM equ_inspect_plan WHERE created_time like '{current_day}%' and actual_done_time is null"
result=iot.select(select_query)
print(result)

# 格式化输出
for code, timestamp in result:
    # 生成随机的分钟数（ 注意数据合理性，分别在7：30和13：30生成记录的，完成时间不要超过当前时间）
    random_minutes = random.randint(40, 180)
    timestamp=timestamp+timedelta(minutes=random_minutes)
    formatted_time = timestamp.strftime('%Y-%m-%d %H:%M:%S')
    print(f'{code}: {formatted_time}')
    sql=f"update equ_inspect_plan set actual_done_time='{formatted_time}',state=9 where wo_no='{code}'"
    iot.dml(sql)